//////////////////////////////////////////////////////////////////////////////////
//                                                                              //
//  This file is part of the buola project (https://code.google.com/p/buola/).  //
//                                                                              //
//  Copyright(c) 2007-2012 Xavi Gratal                                          //
//  gratal AT gmail DOT com                                                     //
//                                                                              //
//  Buola is free software: you can redistribute it and/or modify               //
//  it under the terms of the GNU General Public License as published by        //
//  the Free Software Foundation, either version 3 of the License, or           //
//  (at your option) any later version.                                         //
//                                                                              //
//  Buola is distributed in the hope that it will be useful,                    //
//  but WITHOUT ANY WARRANTY; without even the implied warranty of              //
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the               //
//  GNU General Public License for more details.                                //
//                                                                              //
//  You should have received a copy of the GNU General Public License           //
//  along with buola.  If not, see <http://www.gnu.org/licenses/>.              //
//                                                                              //
//////////////////////////////////////////////////////////////////////////////////

#include <buola/os/user.h>
#include <unistd.h>

namespace buola { namespace os {

CUser get_user()
{
    return CUser(geteuid());
}

CUser get_real_user()
{
    return CUser(getuid());
}

CGroup get_group()
{
    return CGroup(getegid());
}

CGroup get_real_group()
{
    return CGroup(getgid());
}

std::vector<CGroup> get_group_list()
{
    constexpr int lMax=1024>NGROUPS_MAX?NGROUPS_MAX:1024;
    gid_t lList[lMax];

    int lNum=getgroups(lMax,lList);

    std::vector<CGroup> lRet;
    lRet.reserve(lNum+1);

    int lEff=getegid();
    lRet.emplace_back(lEff);
    
    for(int i=0;i<lNum;i++)
    {
        if(lList[i]!=lEff)
            lRet.emplace_back(lList[i]);
    }

    return lRet;
}

bool get_group_membership(CGroup pGroup)
{
    if(pGroup.ID()==getegid()) return true;

    constexpr int lMax=1024>NGROUPS_MAX?NGROUPS_MAX:1024;
    gid_t lList[lMax];

    int lNum=getgroups(lMax,lList);

    for(int i=0;i<lNum;i++)
    {
        if(lList[i]==pGroup.ID())
            return true;
    }

    return false;
}
    
/*namespace os */ } /*namespace buola*/ }
